package com.demo.ai.rag;

import org.springframework.ai.document.Document;

import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;

public class TextSplitter {

    public static List<Document> splitText(String text) {
        // 使用“。”作为分隔符拆分文本
        String[] sentences = text.split("。");

        // 去除空字符串并封装为 Document 对象
        return Arrays.stream(sentences)
                .map(String::trim)
                .filter(s -> !s.isEmpty())
                .map(content -> new Document(content + "。")) // 补回句号
                .collect(Collectors.toList());
    }
}
